
#include "prefs_ee.h"
#include "prefs_ee_config.h"
#include <string.h>

#define EE_ERASE_PAGE_ADDRESS 0
#define EE_ERASE_PAGE_NUMBER 1
#define EE_ERASE_SECTOR_NUMBER 2

#ifdef STM32F0
#define EE_ERASE EE_ERASE_PAGE_ADDRESS
#define FLASH_SIZE ((((uint32_t)(*((uint16_t *)FLASHSIZE_BASE)) & (0xFFFFU))) * 1024)
#endif

#ifdef STM32F1
#define EE_ERASE EE_ERASE_PAGE_ADDRESS
#define FLASH_SIZE ((((uint32_t)(*((uint16_t *)FLASHSIZE_BASE)) & (0xFFFFU))) * 1024)
#endif

#ifdef STM32F2
#define EE_ERASE EE_ERASE_SECTOR_NUMBER
#define FLASH_SIZE ((((uint32_t)(*((uint16_t *)FLASHSIZE_BASE)) & (0xFFFFU))) * 1024)
#endif

#ifdef STM32F3
#define EE_ERASE EE_ERASE_PAGE_ADDRESS
#define FLASH_SIZE ((((uint32_t)(*((uint16_t *)FLASHSIZE_BASE)) & (0xFFFFU))) * 1024)
#endif

#ifdef STM32F4
#define EE_ERASE EE_ERASE_SECTOR_NUMBER
#define FLASH_SIZE ((((uint32_t)(*((uint16_t *)FLASHSIZE_BASE)) & (0xFFFFU))) * 1024)
#endif

#ifdef STM32F7
#define EE_ERASE EE_ERASE_SECTOR_NUMBER
#endif

#ifdef STM32H5
#define EE_ERASE EE_ERASE_SECTOR_NUMBER
#endif

#ifdef STM32H7
#define EE_ERASE EE_ERASE_SECTOR_NUMBER
#endif

#ifdef STM32G0
#define EE_ERASE EE_ERASE_PAGE_NUMBER
#endif

#ifdef STM32G4
#define EE_ERASE EE_ERASE_PAGE_NUMBER
#endif

#ifdef STM32U0
#define EE_ERASE EE_ERASE_PAGE_NUMBER
#endif

#ifdef STM32U5
#define EE_ERASE EE_ERASE_PAGE_NUMBER
#endif

#ifdef STM32L0
#define EE_ERASE EE_ERASE_PAGE_ADDRESS
#endif

#ifdef STM32L1
#define EE_ERASE EE_ERASE_PAGE_ADDRESS
#endif

#ifdef STM32L4
#define EE_ERASE EE_ERASE_PAGE_NUMBER
#endif

#ifdef STM32L5
#define EE_ERASE EE_ERASE_PAGE_NUMBER
#endif

#ifdef STM32WB
#define EE_ERASE EE_ERASE_PAGE_NUMBER
#endif

#ifdef STM32WBA
#define EE_ERASE EE_ERASE_PAGE_NUMBER
#undef FLASH_BANK_1
#endif

#ifdef STM32WL
#define EE_ERASE EE_ERASE_PAGE_NUMBER
#endif

#ifdef STM32C0
#define EE_ERASE EE_ERASE_PAGE_NUMBER
#endif

#ifndef EE_ERASE
#error "Not Supported MCU!"
#endif

EE_HandleTypeDef eeHandle;

/**
 * @brief Initializes the EEPROM emulation module.
 * @note This function initializes the EEPROM emulation module to enable read and write operations.
 * @param pData: Pointer to the start address of the EEPROM emulation area.
 * @param Size: Size of the EEPROM emulation area in bytes.
 * @return Boolean value indicating the success of the initialization:
 *       - true: Initialization successful.
 *       - false: Initialization failed.
 */
bool prefs_ee_init(void *pData, uint32_t Size)
{
    bool answer = false;
    do
    {
        if ((pData == NULL) || (Size == 0))
        {
            break;
        }
#if EE_MANUAL_CONFIG == false
#if (EE_ERASE == EE_ERASE_PAGE_NUMBER) || (EE_ERASE == EE_ERASE_PAGE_ADDRESS)
#ifdef FLASH_PAGE_SIZE
        eeHandle.PageSectorSize = FLASH_PAGE_SIZE;
#endif
#elif (EE_ERASE == EE_ERASE_SECTOR_NUMBER)
#if defined FLASH_SECTOR_SIZE
        eeHandle.PageSectorSize = FLASH_SECTOR_SIZE;
#else
#error EE Library should be set manually for your MCU!
#endif
#endif
#if defined FLASH_BANK_2
        eeHandle.BankNumber = FLASH_BANK_2;
        eeHandle.PageSectorNumber = ((FLASH_SIZE / eeHandle.PageSectorSize / 2) - 1);
        eeHandle.Address = (FLASH_BASE + eeHandle.PageSectorSize * (eeHandle.PageSectorNumber * 2 + 1));
#elif defined FLASH_BANK_1
        eeHandle.BankNumber = FLASH_BANK_1;
        eeHandle.PageSectorNumber = ((FLASH_SIZE / eeHandle.PageSectorSize) - 1);
        eeHandle.Address = (FLASH_BASE + eeHandle.PageSectorSize * eeHandle.PageSectorNumber);
#else
        eeHandle.PageSectorNumber = ((FLASH_SIZE / eeHandle.PageSectorSize) - 1);
        eeHandle.Address = (FLASH_BASE + eeHandle.PageSectorSize * eeHandle.PageSectorNumber);
#endif
#else // manual
#if (defined FLASH_BANK_1) || (defined FLASH_BANK_2)
        eeHandle.BankNumber = EE_SELECTED_BANK;
#endif
        eeHandle.PageSectorNumber = EE_SELECTED_PAGE_SECTOR_NUMBER;
        eeHandle.PageSectorSize = EE_SELECTED_PAGE_SECTOR_SIZE;
        eeHandle.Address = EE_SELECTED_ADDRESS;
#endif
        /* checking size of eeprom area*/
        if (Size > eeHandle.PageSectorSize)
        {
            eeHandle.Size = 0;
            eeHandle.pData = NULL;
            break;
        }
        eeHandle.Size = Size;
        eeHandle.pData = (uint8_t *)pData;
        answer = true;

    } while (0);

    return answer;
}

/**
 * @brief Retrieves the capacity of the EEPROM emulation area.
 * @note This function returns the total capacity of the EEPROM emulation area in bytes.
 * @return Capacity of the EEPROM emulation area in bytes.
 */
uint32_t prefs_ee_capacity(void)
{
    return eeHandle.PageSectorSize;
}

/**
 * @brief Formats the EEPROM emulation area.
 * @note This function formats the EEPROM emulation area,
 * @return bool Boolean value indicating the success of the operation:
 *     - true: Formatting successful.
 *     - false: Formatting failed.
 */
bool prefs_ee_format(void)
{
    bool answer = false;
    uint32_t error;
    FLASH_EraseInitTypeDef flashErase;
    do
    {
        HAL_FLASH_Unlock();
#ifdef HAL_ICACHE_MODULE_ENABLED
        /* disabling ICACHE if enabled*/
        HAL_ICACHE_Disable();
#endif
#if EE_ERASE == EE_ERASE_PAGE_ADDRESS
        flashErase.TypeErase = FLASH_TYPEERASE_PAGES;
        flashErase.PageAddress = eeHandle.Address;
        flashErase.NbPages = 1;
#elif EE_ERASE == EE_ERASE_PAGE_NUMBER
        flashErase.TypeErase = FLASH_TYPEERASE_PAGES;
        flashErase.Page = eeHandle.PageSectorNumber;
        flashErase.NbPages = 1;
#else
        flashErase.TypeErase = FLASH_TYPEERASE_SECTORS;
        flashErase.Sector = eeHandle.PageSectorNumber;
        flashErase.NbSectors = 1;
#endif
#if (defined FLASH_BANK_1) || (defined FLASH_BANK_2)
        flashErase.Banks = eeHandle.BankNumber;
#endif
#ifdef FLASH_VOLTAGE_RANGE_3
        flashErase.VoltageRange = FLASH_VOLTAGE_RANGE_3;
#endif
        /* erasing page/sector */
        if (HAL_FLASHEx_Erase(&flashErase, &error) != HAL_OK)
        {
            break;
        }
        /* checking result */
        if (error != 0xFFFFFFFF)
        {
            break;
        }
        answer = true;

    } while (0);

    HAL_FLASH_Lock();
#ifdef HAL_ICACHE_MODULE_ENABLED
    HAL_ICACHE_Enable();
#endif
    return answer;
}

/**
 * @brief Reads data from the EEPROM emulation area.
 * @note This function reads data from the EEPROM emulation area
 *  and loads it into the specified storage pointer.
 */
void prefs_ee_read(void)
{
    uint8_t *data = eeHandle.pData;
#ifdef HAL_ICACHE_MODULE_ENABLED
    /* disabling ICACHE if enabled*/
    HAL_ICACHE_Disable();
#endif
    if (data != NULL)
    {
        /* reading flash */
        for (uint32_t i = 0; i < eeHandle.Size; i++)
        {
            *data = (*(__IO uint8_t *)(eeHandle.Address + i));
            data++;
        }
    }
#ifdef HAL_ICACHE_MODULE_ENABLED
    /* disabling ICACHE if enabled*/
    HAL_ICACHE_Enable();
#endif
}

/**
 * @brief Writes data to the EEPROM emulation area.
 * @note This function writes data to the EEPROM emulation area.
 * @retval true if the write operation is successful, false otherwise.
 */
bool prefs_ee_write(void)
{
    bool answer = true;
    uint8_t *data = eeHandle.pData;
    do
    {
        /* checking eeprom is initialize correctly */
        if (data == NULL)
        {
            answer = false;
            break;
        }
        /* formating flash area before writing */
        if (prefs_ee_format() == false)
        {
            answer = false;
            break;
        }
        HAL_FLASH_Unlock();
#ifdef HAL_ICACHE_MODULE_ENABLED
        /* disabling ICACHE if enabled*/
        HAL_ICACHE_Disable();
#endif
#if (defined FLASH_TYPEPROGRAM_HALFWORD)
        /* writing buffer to flash */
        for (uint32_t i = 0; i < eeHandle.Size; i += 2)
        {
            uint64_t halfWord;
            memcpy((uint8_t *)&halfWord, data, 2);
            if (HAL_FLASH_Program(FLASH_TYPEPROGRAM_HALFWORD, eeHandle.Address + i, halfWord) != HAL_OK)
            {
                answer = false;
                break;
            }
            data += 2;
        }
#elif (defined FLASH_TYPEPROGRAM_DOUBLEWORD)
        /* writing buffer to flash */
        for (uint32_t i = 0; i < eeHandle.Size; i += 8)
        {
            uint64_t doubleWord;
            memcpy((uint8_t *)&doubleWord, data, 8);
            if (HAL_FLASH_Program(FLASH_TYPEPROGRAM_DOUBLEWORD, eeHandle.Address + i, doubleWord) != HAL_OK)
            {
                answer = false;
                break;
            }
            data += 8;
        }
#elif (defined FLASH_TYPEPROGRAM_QUADWORD)
        /* writing buffer to flash */
        for (uint32_t i = 0; i < eeHandle.Size; i += 16)
        {
            if (HAL_FLASH_Program(FLASH_TYPEPROGRAM_QUADWORD, eeHandle.Address + i, (uint32_t)data) != HAL_OK)
            {
                answer = false;
                break;
            }
            data += 16;
        }
#elif (defined FLASH_TYPEPROGRAM_FLASHWORD)
        /* writing buffer to flash */
        for (uint32_t i = 0; i < eeHandle.Size; i += FLASH_NB_32BITWORD_IN_FLASHWORD * 4)
        {
            if (HAL_FLASH_Program(FLASH_TYPEPROGRAM_FLASHWORD, eeHandle.Address + i, (uint32_t)data) != HAL_OK)
            {
                answer = false;
                break;
            }
            data += FLASH_NB_32BITWORD_IN_FLASHWORD * 4;
        }
#endif
        /* verifying Flash content */
        data = eeHandle.pData;
        for (uint32_t i = 0; i < eeHandle.Size; i++)
        {
            if (*data != (*(__IO uint8_t *)(eeHandle.Address + i)))
            {
                answer = false;
                break;
            }
            data++;
        }

    } while (0);

    HAL_FLASH_Lock();
#ifdef HAL_ICACHE_MODULE_ENABLED
    HAL_ICACHE_Enable();
#endif
    return answer;
}
